"""
Yuxi-Know MCP模块
为知识库系统提供MCP(Model Context Protocol)集成
"""
from server.mcp.mcp_adapter import setup_mcp_server
from server.mcp.yuxi_know_tools import (
    KnowledgeQueryInput, GraphQueryInput, ChatInput, KnowledgeBaseCreateInput,
    EntitySearchInput, SystemHealthOutput, KnowledgeQueryOutput, GraphDataOutput,
    ChatOutput, AvailableKnowledgeBasesOutput, EntitySearchOutput, ToolListOutput,
    GraphStatsOutput
)

__all__ = [
    'setup_mcp_server',
    # 工具输入输出模型
    'KnowledgeQueryInput', 'GraphQueryInput', 'ChatInput', 'KnowledgeBaseCreateInput',
    'EntitySearchInput',
    'SystemHealthOutput', 'KnowledgeQueryOutput', 'GraphDataOutput', 'ChatOutput',
    'AvailableKnowledgeBasesOutput', 'EntitySearchOutput', 'ToolListOutput', 'GraphStatsOutput'
]